package com.example.test.controller;

import com.example.test.bean.UserBean;
import com.example.test.bean.UserImgBean;
import com.example.test.service.UserService;
import com.example.test.utils.FileUtils;
import com.github.pagehelper.PageInfo;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.List;
import java.util.UUID;

@Controller
public class UserController {
    @Resource
    UserService userService;

   // @RequestMapping("/users")
   // public String showUsers(ModelMap modelMap) {
   //     List<UserBean> userList = userService.getAllUser();
    //    modelMap.addAttribute("userList", userList);
     //   return "users";
   // }
    @RequestMapping(value = "/addUser",method = RequestMethod.POST)
    @ResponseBody
    public Map addUser(UserBean userBean, HttpServletRequest request){

        int flag=  userService.insertUser(userBean);
        UserImgBean userImgBean =new UserImgBean();
        List<MultipartFile> files = ((MultipartHttpServletRequest)request).getFiles("file");
        for (MultipartFile multipartFile : files){
            if (multipartFile.isEmpty())
                continue;
            byte[] fileBytes =new byte[0];
            try{
                fileBytes =multipartFile.getBytes();
                String originalFilename =multipartFile.getOriginalFilename();
                originalFilename =
                        originalFilename.substring(originalFilename.lastIndexOf("."));
                String fileName = UUID.randomUUID()+originalFilename;
                boolean bRet = FileUtils.uploadFile(fileBytes,"static/upload/"+fileName);
                if (bRet == false)
                    continue;
                    userImgBean.setId(0);
                    userImgBean.setUser_id(userBean.getId());
                    userImgBean.setImg("upload/"+fileName);
                    int iFlag =userService.insertUserImg(userImgBean);
            }catch (IOException e){
                e.printStackTrace();
            }
        }
        Map<String,Object> map =new HashMap<String,Object>();
        if(flag == 1){
            map.put("msg","新增用户成功");
            return map;
        }else{
            map.put("msg","新增用户失败");
            return  map;
        }
    }
    @RequestMapping(value = "/deleteUser+{id}")
    public String deleteUser(@PathVariable("id") String id,ModelMap modelMap)
    {
           int flag =userService.deleteUser(id);
            List<UserBean> userList =userService.getAllUser();
            modelMap.addAttribute("userList",userList);
            if (flag == 1){
                return "users";
            }else{
                return "error";
        }
    }
   @RequestMapping("/user+{id}")
   public String getUserById(@PathVariable("id") String id,ModelMap modelMap){
        UserBean userBean =userService.getUserById(id);
        List<UserImgBean> listUserImgBean =userService.getUserImgsById(id);
        modelMap.addAttribute("user",userBean);
        modelMap.addAttribute("userImgsList",listUserImgBean);
        return "userInfo";

    }
    @RequestMapping("/modifyUser")
    @ResponseBody
    public Map updateUser(UserBean userBean){
        int flag =userService.updateUser(userBean);
        Map<String,Object> map =new HashMap<>();
        if (flag == 1){
            map.put("msg","修改用户信息成功");
            return map;
        }else{
            map.put("msg","修改用户信息失败");
            return map;
        }
    }
    @RequestMapping("/users")
    public String index(Model model, @RequestParam(defaultValue ="2")
                        Integer pageNum, @RequestParam(defaultValue ="3") Integer pageSize){
        PageInfo<UserBean> pageInfo =userService.selectUser4Page(pageNum,pageSize);
        //使用model传参数回前端
        model.addAttribute("pageInfo",pageInfo);
        model.addAttribute("userList",pageInfo.getList());
        return "users";
    }
}
